package com.leetcode.tree.level_traversal;

import com.leetcode.basic.TreeNode;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * @author Dennis Li
 * @date 2020/6/16 20:20
 */
public class AverageOfLevels_637 {

    // 比较典型的一道BFS的题
    public List<Double> averageOfLevels(TreeNode root) {
        ArrayList<Double> list = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            // 记录这一层的规模
            int size = queue.size();
            // 再备份一个，作为除数，由于结果是double类型，这里用double
            double n = queue.size();
            // 这里有一手阴的，所有值都是int，但是加起来就不一定是int了
            long sum = 0;
            while (size-- > 0) {
                TreeNode node = queue.poll();
                sum += node.val;
                // 分别检测左右子树，非空则入队
                if(node.left != null) queue.offer(node.left);
                if(node.right != null) queue.offer(node.right);
            }
            // 将结果加入到最后的list中
            list.add(sum / n);
        }
        return list;
    }
}
